home *** CD-ROM | disk | FTP | other *** search
/ PC World 2006 February / PCWorld_2006-02_cd.bin / software / vyzkuste / triky / triky.exe / httrack-3.33.exe / {app} / src / htsmodules.c < prev    next >
C/C++ Source or Header  |  2005-02-05  |  12KB  |  404 lines
  1. /* ------------------------------------------------------------ */
  2. /*
  3. HTTrack Website Copier, Offline Browser for Windows and Unix
  4. Copyright (C) Xavier Roche and other contributors
  5.  
  6. This program is free software; you can redistribute it and/or
  7. modify it under the terms of the GNU General Public License
  8. as published by the Free Software Foundation; either version 2
  9. of the License, or any later version.
  10.  
  11. This program is distributed in the hope that it will be useful,
  12. but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14. GNU General Public License for more details.
  15.  
  16. You should have received a copy of the GNU General Public License
  17. along with this program; if not, write to the Free Software
  18. Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19.  
  20.  
  21. Important notes:
  22.  
  23. - We hereby ask people using this source NOT to use it in purpose of grabbing
  24. emails addresses, or collecting any other private information on persons.
  25. This would disgrace our work, and spoil the many hours we spent on it.
  26.  
  27.  
  28. Please visit our Website: http://www.httrack.com
  29. */
  30.  
  31.  
  32. /* ------------------------------------------------------------ */
  33. /* File: htsmodules.c subroutines:                              */
  34. /*       external modules (parsers)                             */
  35. /* Author: Xavier Roche                                         */
  36. /* ------------------------------------------------------------ */
  37.  
  38. /* Internal engine bytecode */
  39. #define HTS_INTERNAL_BYTECODE
  40.  
  41. #include "htsglobal.h"
  42. #include "htsmodules.h"
  43. #include "htsopt.h"
  44. extern int fspc(FILE* fp,char* type);
  45.  
  46. #ifndef _WIN32
  47. #if HTS_DLOPEN
  48. #include <dlfcn.h>
  49. #endif
  50. #endif
  51.  
  52. /* >>> Put all modules definitions here */
  53. #include "htszlib.h"
  54. #include "htsbase.h"
  55.  
  56. typedef int (*t_hts_detect_swf)(htsmoduleStruct* str);
  57. typedef int (*t_hts_parse_swf)(htsmoduleStruct* str);
  58. /* <<< */
  59.  
  60. /* >>> Put all modules includes here */
  61. #include "htsjava.h"
  62. #if HTS_USESWF
  63. #endif
  64. /* <<< */
  65.  
  66. /* >>> Put all modules variables here */
  67.  
  68. int swf_is_available = 0;
  69. t_hts_detect_swf hts_detect_swf = NULL;
  70. t_hts_parse_swf hts_parse_swf = NULL;
  71.  
  72. int gz_is_available = 0;
  73. #if 0
  74. t_gzopen gzopen = NULL;
  75. t_gzread gzread = NULL;
  76. t_gzclose gzclose = NULL;
  77. #endif
  78.  
  79. int SSL_is_available = 0;
  80. t_SSL_shutdown SSL_shutdown = NULL;
  81. t_SSL_free SSL_free = NULL;
  82. t_SSL_CTX_ctrl SSL_CTX_ctrl = NULL;
  83. t_SSL_new SSL_new = NULL;
  84. t_SSL_clear SSL_clear = NULL;
  85. t_SSL_set_fd SSL_set_fd = NULL;
  86. t_SSL_set_connect_state SSL_set_connect_state = NULL;
  87. t_SSL_connect SSL_connect = NULL;
  88. t_SSL_get_error SSL_get_error = NULL;
  89. t_SSL_write SSL_write = NULL;
  90. t_SSL_read SSL_read = NULL;
  91. t_SSL_library_init SSL_library_init = NULL;
  92. t_ERR_load_crypto_strings ERR_load_crypto_strings = NULL;
  93. t_ERR_load_SSL_strings ERR_load_SSL_strings = NULL;
  94. t_SSLv23_client_method SSLv23_client_method = NULL;
  95. t_SSL_CTX_new SSL_CTX_new = NULL;
  96. t_ERR_error_string ERR_error_string = NULL;
  97. t_SSL_load_error_strings SSL_load_error_strings = NULL;
  98.  
  99. int V6_is_available = HTS_INET6;
  100.  
  101. char WHAT_is_available[64]="";
  102. /* <<< */
  103.  
  104. /* memory checks */
  105. HTSEXT_API htsErrorCallback htsCallbackErr = NULL;
  106. HTSEXT_API int htsMemoryFastXfr = 1;    /* fast xfr by default */
  107. void abortLog__fnc(char* msg, char* file, int line);
  108. void abortLog__fnc(char* msg, char* file, int line) {
  109.   FILE* fp = fopen("CRASH.TXT", "wb");
  110.   if (!fp) fp = fopen("/tmp/CRASH.TXT", "wb");
  111.   if (!fp) fp = fopen("C:\\CRASH.TXT", "wb");
  112.   if (!fp) fp = fopen("CRASH.TXT", "wb");
  113.   if (fp) {
  114.     fprintf(fp, "HTTrack " HTTRACK_VERSIONID " closed at '%s', line %d\r\n", file, line);
  115.     fprintf(fp, "Reason:\r\n%s\r\n", msg);
  116.     fflush(fp);
  117.     fclose(fp);
  118.   }
  119. }
  120. HTSEXT_API t_abortLog abortLog__ = abortLog__fnc;    /* avoid VC++ inlining */
  121.  
  122. static void htspe_log(htsmoduleStruct* str, char* msg);
  123.  
  124. int hts_parse_externals(htsmoduleStruct* str) {
  125.   /* >>> Put all module calls here */
  126.  
  127.   /* JAVA */
  128.   if (hts_detect_java(str)) {
  129.     htspe_log(str, "java-lib");
  130.     return hts_parse_java(str);
  131.   }
  132.  
  133. #if HTS_USESWF
  134.   /* FLASH 
  135.    (external module derivated from Macromedia(tm)'s classes) 
  136.   */
  137.   else if (swf_is_available && hts_detect_swf(str)) {
  138.     htspe_log(str, "swf-lib");
  139.     return hts_parse_swf(str);
  140.   }
  141. #endif
  142.  
  143.   /* <<< */
  144.   
  145.   /* Not detected */
  146.   return -1;
  147. }
  148.  
  149. static void addCallback(htscallbacks* chain, void* moduleHandle, htscallbacksfncptr exitFnc) {
  150.   while(chain->next != NULL) {
  151.     chain = chain->next;
  152.   }
  153.   chain->next = calloct(1, sizeof(htscallbacks));
  154.   assertf(chain->next != NULL);
  155.   chain = chain->next;
  156.   memset(chain, 0, sizeof(*chain));
  157.   chain->exitFnc = exitFnc;
  158.   chain->moduleHandle = moduleHandle;
  159. }
  160.  
  161. void clearCallbacks(htscallbacks* chain_);
  162. void clearCallbacks(htscallbacks* chain_) {
  163.   htscallbacks* chain;
  164.   chain = chain_;
  165.   while(chain != NULL) {
  166.     if (chain->exitFnc != NULL) {
  167.       (void) chain->exitFnc();      /* result ignored */
  168.       chain->exitFnc = NULL;
  169.     }
  170.     chain = chain->next;
  171.   }
  172.   chain = chain_;
  173.   while(chain != NULL) {
  174.     if (chain->moduleHandle != NULL) {
  175. #ifdef _WIN32
  176.       FreeLibrary(chain->moduleHandle);
  177. #else
  178.       dlclose(chain->moduleHandle);
  179. #endif
  180.     }
  181.     chain = chain->next;
  182.   }
  183.   chain = chain_->next;     // Don't free the block #0
  184.   while(chain != NULL) {
  185.     htscallbacks* nextchain = chain->next;
  186.     freet(chain);
  187.     chain = nextchain;
  188.   }
  189.   chain_->next = NULL;    // Empty
  190. }
  191.  
  192. void* getFunctionPtr(httrackp* opt, char* file_, char* fncname);
  193. void* getFunctionPtr(httrackp* opt, char* file_, char* fncname) {
  194.   char BIGSTK file[1024];
  195.   void* handle;
  196.   void* userfunction = NULL;
  197.   strcpybuff(file, file_);
  198. #ifdef _WIN32
  199.   handle = LoadLibraryA((char*)file);
  200.   if (handle == NULL) {
  201.     strcatbuff(file, ".dll");
  202.     handle = LoadLibraryA((char*)file);
  203.   }
  204. #else
  205.   handle = dlopen(file, RTLD_LAZY);
  206.   if (handle == NULL) {
  207.     strcatbuff(file, ".so");
  208.     handle = dlopen(file, RTLD_LAZY);
  209.   }
  210. #endif
  211.   if (handle) {
  212.     /* Thanks to Lars Clausen for the "wrapper-init" patch */
  213.     /* If given arguments, call "<wrappername>_init" */
  214.     char BIGSTK tmpName[1024];
  215.     char *comma;
  216.     if ((comma = strchr(fncname, ',')) != NULL) {   /* empty arg */
  217.       *comma++ = '\0';
  218.     }
  219.     
  220.     /* speficic plug init */
  221.     {
  222.       t_htsWrapperPlugInit initfunction;
  223.       sprintf(tmpName, "%s_init", fncname);
  224.       initfunction = (t_htsWrapperPlugInit)DynamicGet(handle, (char*)tmpName);
  225.       if (initfunction  != NULL) {
  226.         int result = (int) initfunction(comma);
  227.         if (!result) {
  228.           if (userfunction == NULL) {
  229. #ifdef _WIN32
  230.             FreeLibrary(handle);
  231. #else
  232.             dlclose(handle);
  233. #endif
  234.           }
  235.           return NULL;
  236.         }       
  237.       }
  238.     }
  239.     /* wrapper_init() */
  240.     {
  241.       t_htsWrapperInit initfunction = (t_htsWrapperInit)DynamicGet(handle, (char*)"wrapper_init");
  242.       if (initfunction != NULL) {
  243.         if (! initfunction(fncname, comma)) {
  244.           if (userfunction == NULL) {
  245. #ifdef _WIN32
  246.             FreeLibrary(handle);
  247. #else
  248.             dlclose(handle);
  249. #endif
  250.           }
  251.           return NULL;
  252.         }
  253.       }
  254.     }
  255.     /* the function itself */
  256.     userfunction = (void*) DynamicGet(handle, (char*)fncname);
  257.     if (userfunction == NULL) {
  258. #ifdef _WIN32
  259.       FreeLibrary(handle);
  260. #else
  261.       dlclose(handle);
  262. #endif
  263.     } else {
  264.       /* optional exit wrapper */
  265.       t_htsWrapperExit exitFnc = (t_htsWrapperExit) DynamicGet(handle, (char*)"wrapper_exit");
  266.       addCallback(&opt->state.callbacks, handle, exitFnc);      // exitFnc can be null
  267.     }
  268.   }
  269.   return userfunction;
  270. }
  271.  
  272. void htspe_init() {
  273.   static int initOk = 0;
  274.   if (!initOk) {
  275.     initOk = 1;
  276.     
  277.     /* >>> Put all module initializations here */
  278.     
  279.  
  280.     /* Zlib */
  281.     gz_is_available = 1;
  282.     /*
  283. #if HTS_DLOPEN
  284.     {
  285.       void* handle;
  286. #ifdef _WIN32
  287.       handle = LoadLibrary("zlib");
  288. #else
  289.       handle = dlopen("libz.so.1", RTLD_LAZY);
  290. #endif
  291.       if (handle) {
  292.         gzopen = (t_gzopen) DynamicGet(handle, "gzopen");
  293.         gzread = (t_gzread) DynamicGet(handle, "gzread");
  294.         gzclose = (t_gzclose) DynamicGet(handle, "gzclose");
  295.         if (gzopen && gzread && gzclose) {
  296.           gz_is_available = 1;
  297.         }
  298.       }
  299.     }
  300. #endif
  301.     */
  302.  
  303.     /* OpenSSL */
  304. #if HTS_DLOPEN
  305.     {
  306.       void* handle;
  307. #ifdef _WIN32
  308.       handle = LoadLibraryA((char*)"ssleay32");
  309. #else
  310.       /* We are compatible with 0.9.6/7 and potentially above */
  311.       handle = dlopen("libssl.so.0.9.7", RTLD_LAZY);
  312.       if (handle == NULL) {
  313.         handle = dlopen("libssl.so.0.9.6", RTLD_LAZY);
  314.       }
  315.       if (handle == NULL) {
  316.         /* Try harder */
  317.         handle = dlopen("libssl.so.0", RTLD_LAZY);
  318.       }
  319. #endif
  320.       if (handle) {
  321.         SSL_shutdown = (t_SSL_shutdown) DynamicGet(handle, (char*)"SSL_shutdown");
  322.         SSL_free = (t_SSL_free) DynamicGet(handle, (char*)"SSL_free");
  323.         SSL_new = (t_SSL_new) DynamicGet(handle, (char*)"SSL_new");
  324.         SSL_clear = (t_SSL_clear) DynamicGet(handle, (char*)"SSL_clear");
  325.         SSL_set_fd = (t_SSL_set_fd) DynamicGet(handle, (char*)"SSL_set_fd");
  326.         SSL_set_connect_state = (t_SSL_set_connect_state) DynamicGet(handle, (char*)"SSL_set_connect_state");
  327.         SSL_connect = (t_SSL_connect) DynamicGet(handle, (char*)"SSL_connect");
  328.         SSL_get_error = (t_SSL_get_error) DynamicGet(handle, (char*)"SSL_get_error");
  329.         SSL_write = (t_SSL_write) DynamicGet(handle, (char*)"SSL_write");
  330.         SSL_read = (t_SSL_read) DynamicGet(handle, (char*)"SSL_read");
  331.         SSL_library_init = (t_SSL_library_init) DynamicGet(handle, (char*)"SSL_library_init");
  332.         ERR_load_SSL_strings = (t_ERR_load_SSL_strings) DynamicGet(handle, (char*)"ERR_load_SSL_strings");
  333.         SSLv23_client_method = (t_SSLv23_client_method) DynamicGet(handle, (char*)"SSLv23_client_method");
  334.         SSL_CTX_new  = (t_SSL_CTX_new) DynamicGet(handle, (char*)"SSL_CTX_new");
  335.         SSL_load_error_strings = (t_SSL_load_error_strings) DynamicGet(handle, (char*)"SSL_load_error_strings");
  336.         SSL_CTX_ctrl = (t_SSL_CTX_ctrl) DynamicGet(handle, (char*)"SSL_CTX_ctrl");
  337. #ifdef _WIN32
  338.         handle = LoadLibraryA((char*)"libeay32");
  339. #endif
  340.         ERR_load_crypto_strings = (t_ERR_load_crypto_strings) DynamicGet(handle, (char*)"ERR_load_crypto_strings");
  341.         ERR_error_string = (t_ERR_error_string) DynamicGet(handle, (char*)"ERR_error_string");
  342.  
  343.         if (SSL_shutdown && SSL_free && SSL_CTX_ctrl && SSL_new && SSL_clear && 
  344.           SSL_set_fd && SSL_set_connect_state && SSL_connect && SSL_get_error && SSL_write 
  345.           && SSL_read && SSL_library_init && SSLv23_client_method && SSL_CTX_new
  346.           && SSL_load_error_strings && ERR_error_string) {
  347.           SSL_is_available = 1;
  348.         }
  349.       }
  350.     }
  351. #endif
  352.     /* */
  353.     
  354.     /* 
  355.     FLASH
  356.     Load the library on-the-fly, if available
  357.     If not, that's not a problem
  358.     */
  359. #if HTS_DLOPEN
  360.     {
  361. #ifdef _WIN32
  362.       void* handle = LoadLibraryA((char*)"htsswf");
  363. #else
  364.       void* handle = dlopen("libhtsswf.so.1", RTLD_LAZY);
  365. #endif
  366.       if (handle) {
  367.         hts_detect_swf = (t_hts_detect_swf) DynamicGet(handle, "hts_detect_swf");
  368.         hts_parse_swf = (t_hts_parse_swf) DynamicGet(handle, "hts_parse_swf");
  369.         if (hts_detect_swf && hts_parse_swf) {
  370.           swf_is_available = 1;
  371.         }
  372.       }
  373.       // FreeLibrary(handle);
  374.       // dlclose(handle);
  375.     }
  376. #endif
  377.  
  378.     /* <<< */
  379.  
  380.     /* Options availability */
  381.     sprintf(WHAT_is_available, "%s%s%s%s",
  382.       V6_is_available ? "" : "-noV6",
  383.       gz_is_available ? "" : "-nozip",
  384.       SSL_is_available ? "" : "-nossl",
  385.       swf_is_available ? "+swf" : "");
  386.  
  387.     
  388.   }
  389. }
  390.  
  391. static void htspe_log(htsmoduleStruct* str, char* msg) {
  392.   char* savename = str->filename;
  393.   httrackp* opt = (httrackp*) str->opt;
  394.   if ((opt->debug>1) && (opt->log!=NULL)) {
  395.     fspc(opt->log,"debug"); fprintf(opt->log,"(External module): parsing %s using module %s"LF, 
  396.       savename, msg);
  397.   }
  398. }
  399.  
  400. HTSEXT_API const char* hts_is_available(void);
  401. HTSEXT_API const char* hts_is_available(void) {
  402.   return WHAT_is_available;
  403. }
  404.